# config.py
import os.path

# gets home dir cross platform
HOME = os.path.expanduser("/home/ubuntu/code/ssd.pytorch")

# for making bounding boxes pretty
COLORS = ((255, 0, 0, 128), (0, 255, 0, 128), (0, 0, 255, 128),
          (0, 255, 255, 128), (255, 0, 255, 128), (255, 255, 0, 128))

# 217.26648179 214.13622267 228.03388083
# MEANS = (104, 117, 123)
# 花粉数据集的means
MEANS = (217, 214, 228)
# filte后的数据集means
# MEANS = (24, 20, 27)
# SSD300 CONFIGS
voc = {
    'num_classes': 6,
    'lr_steps': (8000, 15000, 20000, 24000),
    'max_iter': 24000,
    'feature_maps': [38, 19, 10, 5, 3, 1],
    'min_dim': 300,
    'steps': [8, 16, 32, 64, 100, 300],
    # 'min_sizes': [21, 45, 99, 153, 207, 261],
    # 'max_sizes': [45, 99, 153, 207, 261, 315],
    'min_sizes': [30, 60, 111, 162, 213, 264],
    'max_sizes': [60, 111, 162, 213, 264, 315],
    'aspect_ratios': [[2], [2, 3], [2, 3], [2, 3], [2], [2]],
    # 'aspect_ratios': [[], [], [], [], [], []],
    'variance': [0.1, 0.2],
    'clip': True,
    'name': 'VOC',
}
# voc = {
#     'num_classes': 2,
#     'lr_steps': (5000, 8000, 12000),
#     'max_iter': 13000,
#     'feature_maps': [38, 19, 10, 5, 3, 1],
#     'min_dim': 300,
#     'steps': [8, 16, 32, 64, 100, 300],
#     # 'min_sizes': [21, 45, 99, 153, 207, 261],
#     # 'max_sizes': [45, 99, 153, 207, 261, 315],
#     'min_sizes': [30, 60, 111, 162, 213, 264],
#     'max_sizes': [60, 111, 162, 213, 264, 315],
#     'aspect_ratios': [[2], [2, 3], [2, 3], [2, 3], [2], [2]],
#     # 'aspect_ratios': [[], [], [], [], [], []],
#     'variance': [0.1, 0.2],
#     'clip': True,
#     'name': 'VOC',
# }

voc_custom = {
    'num_classes': 4,
    'lr_steps': (10000,),
    'max_iter': 10000,
    'feature_maps': [38, 19, 10],
    'min_dim': 300,
    'steps': [8, 16, 32],
    'min_sizes': [21, 45, 99],
    'max_sizes': [45, 99, 153],
    # 'min_sizes': [30, 60, 111, 162, 213, 264],
    # 'max_sizes': [60, 111, 162, 213, 264, 315],
    'aspect_ratios': [],
    'variance': [0.1, 0.2],
    'clip': True,
    'name': 'VOC',
}

coco = {
    'num_classes': 201,
    'lr_steps': (280000, 360000, 400000),
    'max_iter': 400000,
    'feature_maps': [38, 19, 10, 5, 3, 1],
    'min_dim': 300,
    'steps': [8, 16, 32, 64, 100, 300],
    'min_sizes': [21, 45, 99, 153, 207, 261],
    'max_sizes': [45, 99, 153, 207, 261, 315],
    'aspect_ratios': [[2], [2, 3], [2, 3], [2, 3], [2], [2]],
    'variance': [0.1, 0.2],
    'clip': True,
    'name': 'COCO',
}
